#! /usr/bin/env python
# -*- coding: utf-8 -*-

import os
from app import create_app,db
from app.models import User,Administrator
from flask_script import Manager,Shell

app = create_app('default')

manager = Manager(app)

def make_shell_context():
    return dict(app=app,db=db,User=User,Administrator=Administrator)

manager.add_command('shell',Shell(make_context = make_shell_context))

@manager.command
def test():
    import unittest
    tests = unittest.TestLoader().discover('tests')
    unittest.TextTestRunner(verbosity=2).run(tests)

if __name__ == '__main__':
    manager.run()
